package com.tanhua.sso.service;

import com.alibaba.dubbo.config.annotation.Reference;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tanhua.common.mapper.UserMapper;
import com.tanhua.common.pojo.User;
import com.tanhua.dubbo.server.api.HuanXinApi;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;


@Service
@Slf4j
public class UserService {

    @Autowired
    private RedisTemplate<String,String> redisTemplate;

    @Autowired
    private UserMapper userMapper;

    @Value("${jwt.secret}")
    private String secret;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    @Reference(version = "1.0.0")
    private HuanXinApi huanXinApi;

    /**
     * 用户登录
     * @param phone 手机号
     * @param code  验证码
     * @return
     */
    public String login(String phone, String code) {
        String redisKey = "CHECK_CODE_"+phone;
        boolean isNew = false;

        //校验验证码
        String redisData = this.redisTemplate.opsForValue().get(redisKey);
        if(!StringUtils.equals(code,redisData)){
            return null;    //验证码错误
        }

        //校验完成，删除验证码
        this.redisTemplate.delete(redisKey);

        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("mobile", phone);

        User user = this.userMapper.selectOne(queryWrapper);

        if(null == user){
            //注册该用户
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md5Hex("123456")); //设置默认密码

            //数据库中不存在，创建新用户
            this.userMapper.insert(user);
            isNew = true;

            //将该用户信息注册到环信平台
            Boolean result = this.huanXinApi.register(user.getId());
            if (!result){
                log.error("注册到环信平台失败！"+user.getId());
            }


        }

        //生成token
        Map<String ,Object> claims = new HashMap<>();
        claims.put("id",user.getId());

        //生成token
        String token = Jwts.builder()
                .setClaims(claims)  //存放安全性要求不高的数据信息
                .signWith(SignatureAlgorithm.HS256, secret) //设置加密方法和加密盐
                .setExpiration(new DateTime().plusHours(12).toDate())    //设置过期时间,12小时
                .compact();

        try {
            Map<String,Object>msg = new HashMap<>();
            msg.put("id",user.getId());
            msg.put("data",System.currentTimeMillis());

            this.rocketMQTemplate.convertAndSend("tanhua-sso-login",msg);
        } catch (MessagingException e) {
            log.error("消息发送失败！",e);
        }
        return token+"|"+isNew;
    }


    public User queryUserByToken(String token) {
        try {
        //通过token解析数据
        Map<String,Object> body = Jwts.parser()
                .setSigningKey(secret)
                .parseClaimsJws(token)
                .getBody();
        User user = new User();
        user.setId(Long.valueOf(body.get("id").toString()));

        //需要返回user对象中的mobile，需要查询数据库获取到mobile数据
        //如果每次都查询数据库，必然会导致性能问题，需要对用户的手机号进行缓存操作
        //数据缓存时，需要设置过期时间，过期时间要与token的时间一致
        //如果用户修改了手机号，需要同步修改redis中的数据

        String redisKey = "TANHUA_USER_MOBILE_"+user.getId();
        if(this.redisTemplate.hasKey(redisKey)){
            String mobile = this.redisTemplate.opsForValue().get(redisKey);
            user.setMobile(mobile);
        }else {
            //查询数据库
            User u = this.userMapper.selectById(user.getId());
            user.setMobile(u.getMobile());

            //手机号写入redis
            //设置JWT中的过期时间：秒
            long timeout = Long.valueOf(body.get("exp").toString()) * 1000 - System.currentTimeMillis();
            this.redisTemplate.opsForValue().set(redisKey,u.getMobile(),timeout, TimeUnit.MILLISECONDS);
        }
        return user;
        }catch (ExpiredJwtException e){
            log.info("token已过期！token = " + token);
        }catch (Exception e){
            log.error("token不合法！ token = " + token,e);
        }
        return null;
    }
}
